from models.labels.keyword_extractor import KeywordExtractor

if __name__ == '__main__':
    model = KeywordExtractor()
    text = '北京时间5月28日消息，据《北京青年报》官微透露，北京中赫国安归化球员李可将入选新一期国家队的大名单。而他将成为国足历史首位归化国脚，目前相关手续应该已经得到落实，意味着李可具备代表中国队参赛的资格。北京时间5月28日消息，据《北京青年报》官微透露，北京中赫国安归化球员李可将入选新一期国家队的大名单。而他将成为国足历史首位归化国脚，目前相关手续应该已经得到落实，意味着李可具备代表中国队参赛的资格。'
    words_list = model.process_text(text)
    tfidf_keyword = model.get_keyword_list(words_list, param='tfidf')
    print('tfidf_keyword', tfidf_keyword)
    textrank_keyword = model.get_keyword_list(words_list, param='textrank')
    print("textrank_keyword", textrank_keyword)
    keyword_interact = model.keyword_interact(tfidf_keyword, textrank_keyword)
    print('keyword_interact', keyword_interact)
    keyword_topk = model.keyword_topk(tfidf_keyword, textrank_keyword, 3)
    print('keyword_topk', keyword_topk)
